package com.bangbang.shiro;

import org.apache.shiro.crypto.SecureRandomNumberGenerator;
import org.apache.shiro.crypto.hash.SimpleHash;

import java.util.LinkedHashMap;
import java.util.Map;

public class ShiroUtils {
    /**
     * 随机生成 salt 需要指定 它的字符串的长度
     *
     * @param len 字符串的长度
     * @return salt
     */
    public static String generateSalt(int len) {
        //一个Byte占两个字节
        int byteLen = len >> 1;
        SecureRandomNumberGenerator secureRandom = new SecureRandomNumberGenerator();
        return secureRandom.nextBytes(byteLen).toHex();
    }

    /**
     * 获取加密后的密码，使用默认hash迭代的次数 1 次
     *
     * @param hashAlgorithm hash算法名称 MD2、MD5、SHA-1、SHA-256、SHA-384、SHA-512、etc。
     * @param password      需要加密的密码
     * @param salt          盐
     * @return 加密后的密码
     */
    public static String encryptPassword(String hashAlgorithm, String password, String salt) {
        return encryptPassword(hashAlgorithm, password, salt, 1);
    }

    /**
     * 获取加密后的密码，需要指定 hash迭代的次数
     *
     * @param hashAlgorithm  hash算法名称 MD2、MD5、SHA-1、SHA-256、SHA-384、SHA-512、etc。
     * @param password       需要加密的密码
     * @param salt           盐
     * @param hashIterations hash迭代的次数
     * @return 加密后的密码
     */
    public static String encryptPassword(String hashAlgorithm, String password, String salt, int hashIterations) {
        SimpleHash hash = new SimpleHash(hashAlgorithm, password, salt, hashIterations);
        return hash.toString();
    }

//    public static void main(String[] args) {
//        String s = ShiroUtils.generateSalt(6);//随机生成6位
//        System.out.println("随机生成一个盐  ----  :    "+s);//生成盐
//        //                                      生成密码加密方式 MD5     名文密码 123       加密4次
//        String md5 = ShiroUtils.encryptPassword("MD5", "123", s, 4);
////        加密出来的名文
//        System.out.println("加密出来的密文是  --  + " + md5);
//    }

    public Map<String,Object> getpwdsalt(String pwd){
        Map<String,Object> map = new LinkedHashMap();
        String s = ShiroUtils.generateSalt(6);//随机生成6位
        map.put("salt",s);
        String md5 = ShiroUtils.encryptPassword("MD5", pwd, s, 4);
        map.put("pwd",md5);
        return map;
    }
}
